from pytorch_widedeep.bayesian_models.tabular import BayesianWide, BayesianTabMlp
